import os.path
from config import screen_pic_FPS, windowname
import cv2 as cv
import numpy as np


def scan_dir(dir_path, pic_type):
    """扫描目录"""
    file_list = []
    for top, dirs, nondirs in os.walk(dir_path):
        for item in nondirs:
            if item.split('.')[-1] == pic_type:
                file_list.append(os.path.join(top, item))
    return file_list


def del_file(path_data):
    """删除文件夹下面的所有文件(只删除文件,不删除文件夹)"""
    for i in os.listdir(path_data):  # os.listdir(path_data)#返回一个列表，里面是当前目录下面的所有东西的相对路径
        file_data = path_data + "\\" + i  # 当前文件夹的下面的所有东西的绝对路径
        # os.path.isfile判断是否为文件,如果是文件,就删除.如果是文件夹.递归给del_file.
        if os.path.isfile(file_data) == True:
            os.remove(file_data)
        else:
            del_file(file_data)


def show_pic(pic_matrix, wait_key=screen_pic_FPS):
    """
    desc: 将pic_matrix展示在屏幕上, 展示时间为wait_key
    param:
    pic_matrix : 照片的数字矩阵
    n: 图像从bmp*n倍显示
    wait_key: 一张图片显示在屏幕的时间
    return:
        NONE
    """
    # # image_data = cv.imread(pic_name)
    # h, w, c = pic_matrix.shape
    # resize_width = w
    # resize_height = h
    # windowname = "Image"    # 使用中文会乱码
    # cv.namedWindow(windowname, cv.WINDOW_NORMAL)
    # cv.resizeWindow(windowname, resize_width, resize_height)
    
    cv.imshow(windowname, pic_matrix)
    cv.waitKey(wait_key)
    # cv.destroyAllWindows()
    # return None
